5.6 Excercises: refactor tests