Updated for consistency with code in the video