Added category images and improved UI