Added some stuff to allow move the balls in real time