Add improved flash