cleaning up lesson 2