Moved plot_weights into plotting
Added return for RBM plot_* functions