Fix bug with #get_image and fix up UI class.