Removing the unnecessary imports