Changed image to init in Unit 1 and 2