Changed the imageView to button