Add flash support