Add onDrag option