Added feature to modify the save button's class

On my app I wanted to use a primary button, not a success button so I
modified this to accept a parameter of a class for the save button