Working on the pose_2d_estimator, not finished!