Updated the code to match the new videos.