Add image tests