Updated image to new Pipeline class