Improve prediction of call/return.

We now model the return stack as part of the prediction process,
making it possible to perform prediction through a call and return
pair.

Fixes #288.