Improves example code in readme