add comments & simplify