add comments and update readme