import model if __name__ == '__main__': if input('Are you sure you want to remove all the banks ownerships? (type in "yes" or something else):') == 'yes': model.reset_bank() else: print('Not resetting bank.') model.cleanup()