Minor fixes to image