Fix example code in readme