Lesson code working fine