Modified to match the lesson