Added a better button