Python Passing Values Between Functions Math