Preparing to add button that saves changes