Modify test from chapter 5 exercises