Cleaning up code snippets in README