training and testing work. need to fix up code and then done.