added other stuff. need to train leaf node module