Update README and make sure everything works