Add a bit of the code to the README