Python Function Returns Two Variables Linear