Python Function Define Input Variable