Updated after watching the vide lectures