Fixed indentation error and removed duplicate 'import sys'