add comments and modify readme