Incorporate img_label better.