Add argument id and time to `get_x_probs`

[helske]

Allows to get only subset of results instead computing everything.

[helske]

Only id for now.

[helske]

It looks like this option does not work as intended, there is a bug somewhere regarding the behaviour of ids argument.