Python Passing Variables Between Functions Python Code