Robot control with augmented reality