Wrote a simple main method in Python.