Add comments fix error in solution