fixed up lesson 1
tidied up lesson 2