Making A Main Function In Python