Don't forget to import sys.