Clean up PlotDrag code