Allow setting the button type