Add imaging code