Added option for button size