Updated README
Code cleanup