Fixing predict and some other things.