hotfixes, all models do "work"

issues with actual prediction, some weird behaviors.