Added import sys for output