Adding updated version of lesson 2