Updating pipeline to work with non-standard models
Updated documentation string