Python file

Moved print statement into main()