Tidy up persistence layer