Almost there...
Just have to fix up 2 things in exercise 6