Issue fixed with course #2