Using os.environ.get('DISPLAY') instead of os.environ['DISPLAY']