Add light ball drag code