Update readme examples

Also making the example a little bit simpler.