Added first main function 'train_rbm(...)'