Adding the UI. Add/Remove working