Updated more of rbm.train.c