Minor changes to baselinelearner