nicer output from predict.py
import fix in lib.py