Python Function Input Variable