Updated some comments and the readme