dev: Tidy up