diff --git a/README.md b/README.md
index ad3c3f8..5511dc8 100644
--- a/README.md
+++ b/README.md
@@ -1,2 +1,2 @@
 # cola-to-nbody
-A suite of codes and pre-trained convolutional neural net models to learn the mapping between the COLA and N-Body simulations of the Universe.
+A suite of codes and pre-trained convolutional neural net models to learn the mapping between the COLA and N-Body simulations of the Universe. The model architecture can be found at https://github.com/eelregit/map2map/blob/master/map2map/models/vnet.py The model has been trained and tested using https://github.com/eelregit/map2map which is a suite of neural network emulators to transform field/map data. For questions/suggestions, please contact me at kaushal@mtu.edu