Removing unnecessary `import sys`