Updated after watching the video lectures