Improve button sizing