Updated the buttons slightly