Problem 6: add a comment
Problem 6: fixes