Partial working solution to drag-and-drop exercise