Remove unnecessary imports