Optimize Function Python Video