Python Function Returns Two Variables Equations