add modefied rbm.py