clean up and comment
Update readme