update readme
code cleanup