Fixed answer in Problem 006