it totally works. commit before machine learning additions