Minor changes to adjust for changed image