Added image code back