Added Save button to controller