Clean up of the code in the README