minor code clean up
Update README.md