Removed unnused import from main