Changed to consistent indent (4)
Added main.py for testing purposes.