Propertly removing drag class after drop.