tidy up import