Python Function Define Input In Math