add comments and code