Tidy up a little, add comments, add readme