small edits to problem 10