re-instated implementation of prediction