Code Cleanup

Removed the extra import as it wasn't needed in the file.