modify to add flash