Training tweaks + estimator stub
First version of the estimator