adding main to python file