Generalizing some more, got some predictions working :)