Writing Main Function In Python