clean up main.py
Merge pull request #1 from nachimath/master

clean up main.py