updated button to be square