Updated __main__.py to add a robot