Added support for probs
Really added support this time.