add save button and functionality