Small bugfixes, the predict part is still not functional, no solution for the multi-class multi-output case. Maybe that's ok.