Minor changes after watching the video