Updated README

Updated some comments in the code and cleanup