Create main.py

Bugs fixed. Moved from earlier ~/passion/ to ~/