added covariance loss experiments, refactored bellman basis object