Tidy up db code