Working on pose_2d_estimator, not finished