Added update function and ball movement.