Enable u-boot fastboot to update all partitions

Change-Id: I2a9fa47d4795632882501ba6c2e855a264e69b8b
diff --git a/flash.sh b/flash.sh
index a79e7ef..b2e9e54 100755
--- a/flash.sh
+++ b/flash.sh
@@ -47,12 +47,12 @@
     load_fastboot "${files_dir}"
 
     try ${FASTBOOT_CMD} erase mmc0
-    try ${FASTBOOT_CMD} erase mmc0boot0
-    try ${FASTBOOT_CMD} erase mmc0boot1
-    try ${FASTBOOT_CMD} flash mmc0 "${files_dir}/partition-table.img"
-    try ${FASTBOOT_CMD} flash mmc0boot0 "${files_dir}/bl2.img"
-    try ${FASTBOOT_CMD} flash mmc0boot1 "${files_dir}/u-boot-env.bin"
   fi
+  try ${FASTBOOT_CMD} erase mmc0boot0
+  try ${FASTBOOT_CMD} erase mmc0boot1
+  try ${FASTBOOT_CMD} flash mmc0 "${files_dir}/partition-table.img"
+  try ${FASTBOOT_CMD} flash mmc0boot0 "${files_dir}/bl2.img"
+  try ${FASTBOOT_CMD} flash mmc0boot1 "${files_dir}/u-boot-env.bin"
   try ${FASTBOOT_CMD} flash bootloaders "${files_dir}/fip.bin"
   try ${FASTBOOT_CMD} flash boot "${files_dir}/boot_arm64.img"
   try ${FASTBOOT_CMD} flash rootfs "${files_dir}/rootfs_arm64.img"
@@ -151,6 +151,7 @@
   -s <serial>         only flashes the board with the given fastboot serial number
   -r <detect_retries> number of times to retry waiting for a device (defaults to 0)
   -o                  flash the old boards.
+  -u                  flash with u-boot fastboot.
 EOF
 )
   local detect_retries=1          # -r <retry_count>