Cleanup README example code a bit

So that it easier to read on github.