Infer latent representations again from latent vectors
Add MDS to visualize the feature space
Add a feature visualizer
Modularize