Added a way to set primary button