Minor updates to imaging scripts