Add flash functionality