Return instead of sys.exit(0)