Add save button (Closes #1)