Added probability to the console.

Added probability to the console and added few more training sets.