start to add comments, not quite working properly