diff --git a/flashutils/flashutils.h b/flashutils/flashutils.h
index 953dd00..d5dadcb 100644
--- a/flashutils/flashutils.h
+++ b/flashutils/flashutils.h
@@ -1,3 +1,6 @@
+#ifndef FLASHUTILS_H
+#define FLASHUTILS_H
+
 int restore_raw_partition(const char *partition, const char *filename);
 int backup_raw_partition(const char *partition, const char *filename);
 int erase_raw_partition(const char *partition);
@@ -35,4 +38,14 @@
 extern int cmd_bml_mount_partition(const char *partition, const char *mount_point, const char *filesystem, int read_only);
 extern int cmd_bml_get_partition_device(const char *partition, char *device);
 
+extern int device_flash_type();
 
+enum flash_type {
+    UNSUPPORTED = -1,
+    UNKNOWN = 0,
+    MTD = 1,
+    MMC = 2,
+    BML = 3
+};
+
+#endif
\ No newline at end of file