Add imaging test