beginnings of machine learning application
adds saving X and y functionality