update code in readme. Wish there was a way to reference.