Cleaned up a code snippet in the README