Changed 'button' back to 'div'