Adding import statement for sys