Section 5.6 exercise improve tests