Corrected readme example

I've added two lines which I found necessary to get the example to run.