Python Assign Variable To Function Return