Update README for new code

This updates the readme to match the new code that I've added.