added “prediction” method, but still needs some work