Removed unnecessary imports and plot() argument