Got decision trees working, need better rules