exercise 2 working but not great