Fix the delete, add, and save button

Make sure that they are functional with the new UI introduced