Did a little cleaning up and commenting. Also adding training data to the repo.