Module 2 fixing silly mistake