Removed unnecessary import and comment