fixed small error, indentations, clarify comments