Update readme
Code cleanup