add rbm.py