Update Main.py

change flash to only the given screen