Fix code snippet in readme