Add button class to save button