Make the code in the README actually work. Derp.