Tidy up API