API tidy up