cleaned some of lesson code