readme edit
code cleanup