Add test for flash