DOC Better documentation
ENH Add switch for prediction results