Added RBM class and training code for test