Added aesthetics, save button