minor cleanup for problem 5