Tidy up naming