slight fix to ch2 exercise