Update the readme with comments