Defining A Main Function Python