cleaned up rbm-stability notebook