Deleted unnecessary class import