small fix to machine learning helper methods. some small edits to prepare for machine learning exploration.