Set onClick handler on button container