Python Function Input Variable And Output