(#34) Added the save button and functionality