Improve specs for class 2 exercise 6