Add image checking; demo functionality