diff --git a/Makefile b/Makefile
index 23e3e93..3641bcc 100644
--- a/Makefile
+++ b/Makefile
@@ -1,4 +1,5 @@
 TARGET = iphone:11.2:10.0
+ARCHS := arm64
 
 include $(THEOS)/makefiles/common.mk
 
diff --git a/Tweak.xm b/Tweak.xm
index 1728ad0..d9bb676 100644
--- a/Tweak.xm
+++ b/Tweak.xm
@@ -47,31 +47,35 @@ extern "C" kern_return_t IOMobileFramebufferSwapSetLayer(
     int flags
 );
 
+extern "C" void IOSurfaceFlushProcessorCaches(IOSurfaceRef buffer);
+extern "C" int IOSurfaceLock(IOSurfaceRef surface, uint32_t options, uint32_t *seed);
+extern "C" int IOSurfaceUnlock(IOSurfaceRef surface, uint32_t options, uint32_t *seed);
+
 int bmp_write(const void *image, size_t xsize, size_t ysize, const char *filename) {
-    unsigned char header[54] = {
-        0x42, 0x4d, 0, 0, 0, 0, 0, 0, 0, 0,
-        54, 0, 0, 0, 40, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 32, 0,
-        0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
-        0, 0, 0, 0
-    };
+    // unsigned char header[54] = {
+    //     0x42, 0x4d, 0, 0, 0, 0, 0, 0, 0, 0,
+    //     54, 0, 0, 0, 40, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 32, 0,
+    //     0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
+    //     0, 0, 0, 0
+    // };
 
-    long file_size = (long)xsize * (long)ysize * 4 + 54;
-    header[2] = (unsigned char)(file_size &0x000000ff);
-    header[3] = (file_size >> 8) & 0x000000ff;
-    header[4] = (file_size >> 16) & 0x000000ff;
-    header[5] = (file_size >> 24) & 0x000000ff;
+    // long file_size = (long)xsize * (long)ysize * 4 + 54;
+    // header[2] = (unsigned char)(file_size &0x000000ff);
+    // header[3] = (file_size >> 8) & 0x000000ff;
+    // header[4] = (file_size >> 16) & 0x000000ff;
+    // header[5] = (file_size >> 24) & 0x000000ff;
 
-    long width = xsize;
-    header[18] = width & 0x000000ff;
-    header[19] = (width >> 8) &0x000000ff;
-    header[20] = (width >> 16) &0x000000ff;
-    header[21] = (width >> 24) &0x000000ff;
+    // long width = xsize;
+    // header[18] = width & 0x000000ff;
+    // header[19] = (width >> 8) &0x000000ff;
+    // header[20] = (width >> 16) &0x000000ff;
+    // header[21] = (width >> 24) &0x000000ff;
 
-    long height = ysize;
-    header[22] = height &0x000000ff;
-    header[23] = (height >> 8) &0x000000ff;
-    header[24] = (height >> 16) &0x000000ff;
-    header[25] = (height >> 24) &0x000000ff;
+    // long height = ysize;
+    // header[22] = height &0x000000ff;
+    // header[23] = (height >> 8) &0x000000ff;
+    // header[24] = (height >> 16) &0x000000ff;
+    // header[25] = (height >> 24) &0x000000ff;
 
     char fname_bmp[128];
     sprintf(fname_bmp, "%s", filename);
@@ -82,13 +86,31 @@ int bmp_write(const void *image, size_t xsize, size_t ysize, const char *filenam
         return -1;
     }
 
-    fwrite(header, sizeof(unsigned char), 54, fp);
+    // fwrite(header, sizeof(unsigned char), 54, fp);
     fwrite(image, sizeof(unsigned char), (size_t)(long)xsize * ysize * 4, fp);
 
     fclose(fp);
     return 0;
 }
 
+int write_to_file(const void *image, size_t xsize, size_t ysize, size_t pixel_size, const char *filename) {
+    char fname_bmp[128];
+    sprintf(fname_bmp, "%s", filename);
+    if( access( fname_bmp, F_OK ) != -1 ) {
+        return 0;
+    } else {
+        FILE *fp;
+        if (!(fp = fopen(fname_bmp, "wb"))) {
+            NSLog(@"Error no is : %s, %d", fname_bmp, errno);
+            return -1;
+        }
+        NSLog(@"sharat write to file");
+        fwrite(image, sizeof(unsigned char), (size_t)(long)xsize * ysize * pixel_size, fp);
+        fclose(fp);
+    }
+    return 0;
+}
+
 %hookf(kern_return_t, IOMobileFramebufferSwapSetLayer, IOMobileFramebufferRef fb, int layer, IOSurfaceRef buffer, CGRect bounds, CGRect frame, int flags) {
     CGSize size;
     size_t width_;
@@ -98,13 +120,21 @@ int bmp_write(const void *image, size_t xsize, size_t ysize, const char *filenam
     height_ = size.height;
     size_t width = IOSurfaceGetWidth(buffer);
     size_t height = IOSurfaceGetHeight(buffer);
-    NSLog(@"sharat %ld, %ld, %ld, %ld", width_, height_, width, height);
+    size_t byte_per_pixel = IOSurfaceGetBytesPerElement(buffer);
+    NSLog(@"sharat %ld, %ld, %ld, %ld, %ld", width_, height_, width, height, byte_per_pixel);
     NSString *path = @"/tmp/test.bmp";
     void *bytes = IOSurfaceGetBaseAddress(buffer);
-    if(width) {
-        int ret;
-        ret = bmp_write(bytes, width, height, [path UTF8String]);
-        NSLog(@"sharat %d", ret);
-    }
+    // if(width) {
+    //     int ret;
+    //     ret = bmp_write(bytes, width, height, [path UTF8String]);
+    //     NSLog(@"sharat %d", ret);
+    // }
+
+    IOSurfaceLock(buffer, kIOSurfaceLockReadOnly, NULL);
+    IOSurfaceFlushProcessorCaches(buffer);
+    write_to_file(bytes, width, height, byte_per_pixel, [path UTF8String]);
+    IOSurfaceUnlock(buffer, kIOSurfaceLockReadOnly, NULL);
+
     return %orig;
 }
+
diff --git a/control b/control
index b24cf02..ff10470 100644
--- a/control
+++ b/control
@@ -1,4 +1,4 @@
-Package: com.appknox.screendump
+Package: com.cosmosgenius.screendump
 Name: screendump
 Depends: mobilesubstrate
 Version: 0.0.1