Python Assign Function To Variable