added tests and simple preds