added model prediction and simple validation