Add save button to UI
Improve dragging and dropping visualisation