Remove unnecesary imports