Added the main function Python trick