Add onClick to image and update button css