Update README.md

Update for cleaned up code