Move registering the image to its own step