Python Function Input String Or List