Made some small changes to the buttons