added save button, functionality TODO