Tidy up before release

Closes #2 as we'll be addressing any issues here.