Uppdaterad med kod för lektion 7