Added IPython notebook, fixed axes handling in proba_test.