Add flash function