Update rbm.py