Assign Function To Variable Python