Fix code in README

Closes #1
Update README