Added a check to set size based on the shape of the button