Update images and add test