Clean up and readme