Added callback for onDragEnd. WIP