Added comment, fixed small error