Removing unused import of sys