flash_mem and flash_test code formatting

flash_mem and flash_test code formatting