small edits to problem 6