Python Assign Variable In Function