Update flash.py

updated comments