Tidy up + doc