Move image into controller