Making class 1 of prob. more concise.

Signed-off-by: Eduardo Tenorio <embatbr@gmail.com>