Added comments
fixed problem 12