def main(system, ai, mode):
"""
TrafficAI command line tool
"""
controller = Controller()
controller.load_system(system, ai)
if mode == 'run':
controller.present()
elif mode == 'train':
controller.develop()