everything works

now just the decision tree to do