Cleaned up the example code in the README.