Improve specs for class 2 exercise 4