Remove unecesary imports