cleanup imports and mains