Add import sys function