updated main.py

now it waits until the ball is within some predicted arrival time of the
robot -s