Updated comments, still working on training data