Fix code snippets in readme