Made the remove buttons work (UI)