Remove unecessary imports