diff --git a/requirements.yml b/requirements.yml
index 551dd90..3785240 100755
--- a/requirements.yml
+++ b/requirements.yml
@@ -1,337 +1,327 @@
-name: brainmage
+name: brainmage_noversion
 channels:
   - pytorch
   - defaults
 dependencies:
-  - _anaconda_depends=2020.02=py36_0
-  - _libgcc_mutex=0.1=main
-  - alabaster=0.7.12=py36_0
-  - anaconda=custom=py36_1
-  - anaconda-client=1.7.2=py36_0
-  - anaconda-project=0.8.4=py_0
-  - argh=0.26.2=py36_0
-  - asn1crypto=1.3.0=py36_0
-  - astroid=2.3.3=py36_0
-  - astropy=4.0.1.post1=py36h7b6447c_1
-  - atomicwrites=1.4.0=py_0
-  - attrs=19.3.0=py_0
-  - autopep8=1.4.3=py36_0
-  - babel=2.8.0=py_0
-  - backcall=0.2.0=py_0
-  - backports=1.0=py_2
-  - backports.shutil_get_terminal_size=1.0.0=py36_2
-  - beautifulsoup4=4.9.1=py36_0
-  - bitarray=1.2.2=py36h7b6447c_0
-  - bkcharts=0.2=py36_0
-  - blas=1.0=mkl
-  - bleach=3.1.5=py_0
-  - blosc=1.19.0=hd408876_0
-  - bokeh=2.1.0=py36_0
-  - boto=2.49.0=py36_0
-  - bottleneck=1.3.2=py36heb32a55_1
-  - brotlipy=0.7.0=py36h7b6447c_1000
-  - bzip2=1.0.8=h7b6447c_0
-  - ca-certificates=2020.1.1=0
-  - cairo=1.14.12=h8948797_3
-  - certifi=2020.6.20=py36_0
-  - cffi=1.14.0=py36h2e261b9_0
-  - chardet=3.0.4=py36_1003
-  - click=7.1.2=py_0
-  - cloudpickle=1.4.1=py_0
-  - clyent=1.2.2=py36_1
-  - colorama=0.4.3=py_0
-  - contextlib2=0.6.0.post1=py_0
-  - contextvars=2.4=py_0
-  - cryptography=2.9.2=py36h1ba5d50_0
-  - cudatoolkit=9.2=0
-  - curl=7.69.1=hbc83047_0
-  - cycler=0.10.0=py36_0
-  - cython=0.29.20=py36he6710b0_0
-  - cytoolz=0.10.1=py36h7b6447c_0
-  - dask=2.19.0=py_0
-  - dask-core=2.19.0=py_0
-  - dbus=1.13.14=hb2f20db_0
-  - decorator=4.4.2=py_0
-  - defusedxml=0.6.0=py_0
-  - diff-match-patch=20181111=py_0
-  - distributed=2.19.0=py36_0
-  - docutils=0.16=py36_1
-  - entrypoints=0.3=py36_0
-  - et_xmlfile=1.0.1=py36_0
-  - expat=2.2.6=he6710b0_0
-  - fastcache=1.1.0=py36h7b6447c_0
-  - filelock=3.0.12=py_0
-  - flake8=3.7.9=py36_0
-  - flask=1.1.2=py_0
-  - fontconfig=2.13.0=h9420a91_0
-  - freetype=2.10.2=h5ab3b9f_0
-  - fribidi=1.0.5=h7b6447c_0
-  - fsspec=0.7.4=py_0
-  - future=0.18.2=py36_1
-  - get_terminal_size=1.0.0=haa9412d_0
-  - gevent=1.4.0=py36h7b6447c_0
-  - glib=2.63.1=h5a9c865_0
-  - glob2=0.7=py_0
-  - gmp=6.1.2=h6c8ec71_1
-  - gmpy2=2.0.8=py36h10f8cd9_2
-  - graphite2=1.3.13=h23475e2_0
-  - greenlet=0.4.16=py36h7b6447c_0
-  - gst-plugins-base=1.14.0=hbbd80ab_1
-  - gstreamer=1.14.0=hb453b48_1
-  - h5py=2.10.0=py36h7918eee_0
-  - harfbuzz=1.8.8=hffaf4a1_0
-  - hdf5=1.10.4=hb1b8bf9_0
-  - heapdict=1.0.1=py_0
-  - html5lib=1.0.1=py36_0
-  - hypothesis=5.16.1=py_0
-  - icu=58.2=he6710b0_3
-  - idna=2.9=py_1
-  - imageio=2.8.0=py_0
-  - imagesize=1.2.0=py_0
-  - immutables=0.14=py36h7b6447c_0
-  - importlib-metadata=1.6.1=py36_0
-  - importlib_metadata=1.6.1=0
-  - intel-openmp=2020.1=217
-  - intervaltree=3.0.2=py_0
-  - ipykernel=5.3.0=py36h5ca1d4c_0
-  - ipython=7.15.0=py36_0
-  - ipython_genutils=0.2.0=py36_0
-  - ipywidgets=7.5.1=py_0
-  - isort=4.3.21=py36_0
-  - itsdangerous=1.1.0=py36_0
-  - jbig=2.1=hdba287a_0
-  - jdcal=1.4.1=py_0
-  - jedi=0.15.2=py36_0
-  - jeepney=0.4.3=py_0
-  - jinja2=2.11.2=py_0
-  - joblib=0.15.1=py_0
-  - jpeg=9b=h024ee3a_2
-  - json5=0.9.5=py_0
-  - jsonschema=3.2.0=py36_0
-  - jupyter=1.0.0=py36_7
-  - jupyter_client=6.1.3=py_0
-  - jupyter_console=6.1.0=py_0
-  - jupyter_core=4.6.3=py36_0
-  - jupyterlab=2.1.4=py_1
-  - jupyterlab_server=1.1.5=py_0
-  - keyring=21.1.1=py36_2
-  - kiwisolver=1.2.0=py36hfd86e86_0
-  - krb5=1.17.1=h173b8e3_0
-  - lazy-object-proxy=1.5.0=py36h7b6447c_0
-  - ld_impl_linux-64=2.33.1=h53a641e_7
-  - libarchive=3.4.2=h62408e4_0
-  - libcurl=7.69.1=h20c2e04_0
-  - libedit=3.1.20191231=h7b6447c_0
-  - libffi=3.2.1=hd88cf55_4
-  - libgcc-ng=9.1.0=hdf63c60_0
-  - libgfortran-ng=7.3.0=hdf63c60_0
-  - liblief=0.10.1=he6710b0_0
-  - libpng=1.6.37=hbc83047_0
-  - libsodium=1.0.16=h1bed415_0
-  - libspatialindex=1.9.3=he6710b0_0
-  - libssh2=1.9.0=h1ba5d50_1
-  - libstdcxx-ng=9.1.0=hdf63c60_0
-  - libtiff=4.1.0=h2733197_1
-  - libtool=2.4.6=h7b6447c_5
-  - libuuid=1.0.3=h1bed415_2
-  - libxcb=1.13=h1bed415_1
-  - libxml2=2.9.10=he19cac6_1
-  - llvmlite=0.32.1=py36hd408876_0
-  - locket=0.2.0=py36_1
-  - lxml=4.5.1=py36hefd8a0e_0
-  - lz4-c=1.9.2=he6710b0_0
-  - lzo=2.10=h7b6447c_2
-  - markupsafe=1.1.1=py36h7b6447c_0
-  - matplotlib=3.2.2=0
-  - matplotlib-base=3.2.2=py36hef1b27d_0
-  - mccabe=0.6.1=py36_1
-  - mistune=0.8.4=py36h7b6447c_0
-  - mkl=2020.1=217
-  - mkl-service=2.3.0=py36he904b0f_0
-  - mkl_fft=1.1.0=py36h23d657b_0
-  - mkl_random=1.1.1=py36h0573a6f_0
-  - mock=4.0.2=py_0
-  - more-itertools=8.4.0=py_0
-  - mpc=1.1.0=h10f8cd9_1
-  - mpfr=4.0.1=hdf1c602_3
-  - mpmath=1.1.0=py36_0
-  - msgpack-python=1.0.0=py36hfd86e86_1
-  - multipledispatch=0.6.0=py36_0
-  - nbconvert=5.6.1=py36_0
-  - nbformat=5.0.7=py_0
-  - ncurses=6.2=he6710b0_1
-  - networkx=2.4=py_0
-  - ninja=1.9.0=py36hfd86e86_0
-  - nltk=3.5=py_0
-  - nose=1.3.7=py36_2
-  - notebook=6.0.3=py36_0
-  - numba=0.49.1=py36h0573a6f_0
-  - numexpr=2.7.1=py36h423224d_0
-  - numpy=1.18.5=py36ha1c710e_0
-  - numpy-base=1.18.5=py36hde5b4d6_0
-  - numpydoc=1.0.0=py_0
-  - olefile=0.46=py36_0
-  - openpyxl=3.0.3=py_0
-  - openssl=1.1.1g=h7b6447c_0
-  - packaging=20.4=py_0
-  - pandas=1.0.5=py36h0573a6f_0
-  - pandoc=2.2.3.2=0
-  - pandocfilters=1.4.2=py36_1
-  - pango=1.42.4=h049681c_0
-  - parso=0.5.2=py_0
-  - partd=1.1.0=py_0
-  - patchelf=0.11=he6710b0_0
-  - path=13.1.0=py36_0
-  - path.py=12.4.0=0
-  - pathlib2=2.3.5=py36_0
-  - pathtools=0.1.2=py_1
-  - patsy=0.5.1=py36_0
-  - pcre=8.43=he6710b0_0
-  - pep8=1.7.1=py36_0
-  - pexpect=4.8.0=py36_0
-  - pickleshare=0.7.5=py36_0
-  - pillow=7.1.2=py36hb39fc2d_0
-  - pip=20.1.1=py36_1
-  - pixman=0.38.0=h7b6447c_0
-  - pkginfo=1.5.0.1=py36_0
-  - pluggy=0.13.1=py36_0
-  - ply=3.11=py36_0
-  - prometheus_client=0.8.0=py_0
-  - prompt-toolkit=3.0.5=py_0
-  - prompt_toolkit=3.0.5=0
-  - psutil=5.7.0=py36h7b6447c_0
-  - ptyprocess=0.6.0=py36_0
-  - py=1.8.2=py_0
-  - py-lief=0.10.1=py36h403a769_0
-  - pycodestyle=2.5.0=py36_0
-  - pycosat=0.6.3=py36h7b6447c_0
-  - pycparser=2.20=py_0
-  - pycrypto=2.6.1=py36h7b6447c_10
-  - pycurl=7.43.0.5=py36h1ba5d50_0
-  - pydocstyle=5.0.2=py_0
-  - pyflakes=2.1.1=py36_0
-  - pygments=2.6.1=py_0
-  - pylint=2.4.4=py36_0
-  - pyodbc=4.0.30=py36he6710b0_0
-  - pyopenssl=19.1.0=py36_0
-  - pyparsing=2.4.7=py_0
-  - pyqt=5.9.2=py36h05f1152_2
-  - pyrsistent=0.16.0=py36h7b6447c_0
-  - pysocks=1.7.1=py36_0
-  - pytables=3.6.1=py36h71ec239_0
-  - pytest=5.4.3=py36_0
-  - pytest-arraydiff=0.3=py36h39e3cac_0
-  - pytest-astropy=0.8.0=py_0
-  - pytest-astropy-header=0.1.2=py_0
-  - pytest-doctestplus=0.7.0=py_0
-  - pytest-openfiles=0.5.0=py_0
-  - pytest-remotedata=0.3.2=py36_0
-  - python=3.6.10=hcf32534_1
-  - python-dateutil=2.8.1=py_0
-  - python-jsonrpc-server=0.3.4=py_0
-  - python-language-server=0.31.10=py36_0
-  - python-libarchive-c=2.9=py_0
-  - pytorch=1.5.1=py3.6_cuda9.2.148_cudnn7.6.3_0
-  - pytz=2020.1=py_0
-  - pywavelets=1.1.1=py36h7b6447c_0
-  - pyxdg=0.26=py_0
-  - pyyaml=5.3.1=py36h7b6447c_0
-  - pyzmq=18.1.1=py36he6710b0_0
-  - qdarkstyle=2.8.1=py_0
-  - qt=5.9.7=h5867ecd_1
-  - qtawesome=0.7.2=py_0
-  - qtconsole=4.7.5=py_0
-  - qtpy=1.9.0=py_0
-  - readline=8.0=h7b6447c_0
-  - regex=2020.5.14=py36h7b6447c_0
-  - requests=2.24.0=py_0
-  - ripgrep=11.0.2=he32d670_0
-  - rope=0.17.0=py_0
-  - rtree=0.9.4=py36_1
-  - ruamel_yaml=0.15.87=py36h7b6447c_0
-  - scikit-image=0.16.2=py36h0573a6f_0
-  - scikit-learn=0.23.1=py36h423224d_0
-  - scipy=1.4.1=py36h0b6359f_0
-  - seaborn=0.10.1=py_0
-  - secretstorage=3.1.2=py36_0
-  - send2trash=1.5.0=py36_0
-  - setuptools=47.3.1=py36_0
-  - simplegeneric=0.8.1=py36_2
-  - singledispatch=3.4.0.3=py36_0
-  - sip=4.19.8=py36hf484d3e_0
-  - six=1.15.0=py_0
-  - snappy=1.1.7=hbae5bb6_3
-  - snowballstemmer=2.0.0=py_0
-  - sortedcollections=1.2.1=py_0
-  - sortedcontainers=2.2.2=py_0
-  - soupsieve=2.0.1=py_0
-  - sphinx=3.1.1=py_0
-  - sphinxcontrib=1.0=py36_1
-  - sphinxcontrib-applehelp=1.0.2=py_0
-  - sphinxcontrib-devhelp=1.0.2=py_0
-  - sphinxcontrib-htmlhelp=1.0.3=py_0
-  - sphinxcontrib-jsmath=1.0.1=py_0
-  - sphinxcontrib-qthelp=1.0.3=py_0
-  - sphinxcontrib-serializinghtml=1.1.4=py_0
-  - sphinxcontrib-websupport=1.2.2=py_0
-  - spyder=4.1.3=py36_0
-  - spyder-kernels=1.9.1=py36_0
-  - sqlalchemy=1.3.17=py36h7b6447c_0
-  - sqlite=3.32.3=h62c20be_0
-  - statsmodels=0.11.1=py36h7b6447c_0
-  - sympy=1.6=py36_0
-  - tbb=2020.0=hfd86e86_0
-  - tblib=1.6.0=py_0
-  - terminado=0.8.3=py36_0
-  - testpath=0.4.4=py_0
-  - threadpoolctl=2.1.0=pyh5ca1d4c_0
-  - tk=8.6.10=hbc83047_0
-  - toolz=0.10.0=py_0
-  - torchvision=0.6.1=py36_cu92
-  - tornado=6.0.4=py36h7b6447c_1
-  - tqdm=4.46.1=py_0
-  - traitlets=4.3.3=py36_0
-  - typed-ast=1.4.1=py36h7b6447c_0
-  - typing_extensions=3.7.4.2=py_0
-  - ujson=1.35=py36h14c3975_0
-  - unicodecsv=0.14.1=py36_0
-  - unixodbc=2.3.7=h14c3975_0
-  - urllib3=1.25.9=py_0
-  - watchdog=0.10.2=py36_0
-  - wcwidth=0.2.4=py_0
-  - webencodings=0.5.1=py36_1
-  - werkzeug=1.0.1=py_0
-  - wheel=0.34.2=py36_0
-  - widgetsnbextension=3.5.1=py36_0
-  - wrapt=1.12.1=py36h7b6447c_1
-  - wurlitzer=2.0.0=py36_0
-  - xlrd=1.2.0=py36_0
-  - xlsxwriter=1.2.9=py_0
-  - xlwt=1.3.0=py36_0
-  - xz=5.2.5=h7b6447c_0
-  - yaml=0.1.7=had09818_2
-  - yapf=0.29.0=py_0
-  - zeromq=4.3.1=he6710b0_3
-  - zict=2.0.0=py_0
-  - zipp=3.1.0=py_0
-  - zlib=1.2.11=h7b6447c_3
-  - zstd=1.4.4=h0b5b093_3
+  - _anaconda_depends
+  - _libgcc_mutex
+  - alabaster
+  - anaconda
+  - anaconda-client
+  - anaconda-project
+  - argh
+  - asn1crypto
+  - astroid
+  - astropy
+  - atomicwrites
+  - attrs
+  - autopep8
+  - babel
+  - backcall
+  - backports
+  - backports.shutil_get_terminal_size
+  - beautifulsoup4
+  - bitarray
+  - bkcharts
+  - blas
+  - bleach
+  - blosc
+  - bokeh
+  - boto
+  - bottleneck
+  - brotlipy
+  - bzip2
+  - ca-certificates
+  - cairo
+  - certifi
+  - cffi
+  - chardet
+  - click
+  - cloudpickle
+  - clyent
+  - colorama
+  - contextlib2
+  - contextvars
+  - cryptography
+  - cudatoolkit
+  - curl
+  - cycler
+  - cython
+  - cytoolz
+  - dask
+  - dask-core
+  - decorator
+  - defusedxml
+  - diff-match-patch
+  - distributed
+  - docutils
+  - entrypoints
+  - et_xmlfile
+  - expat
+  - fastcache
+  - filelock
+  - flake8
+  - flask
+  - fontconfig
+  - freetype
+  - fribidi
+  - fsspec
+  - future
+  - get_terminal_size
+  - gevent
+  - glib
+  - glob2
+  - gmpy2
+  - graphite2
+  - greenlet
+  - gst-plugins-base
+  - gstreamer
+  - h5py
+  - harfbuzz
+  - hdf5
+  - heapdict
+  - html5lib
+  - hypothesis
+  - icu
+  - idna
+  - imageio
+  - imagesize
+  - immutables
+  - importlib-metadata
+  - importlib_metadata
+  - intel-openmp
+  - intervaltree
+  - ipykernel
+  - ipython
+  - ipython_genutils
+  - ipywidgets
+  - isort
+  - itsdangerous
+  - jdcal
+  - jedi
+  - jeepney
+  - jinja2
+  - joblib
+  - jpeg
+  - json5
+  - jsonschema
+  - jupyter
+  - jupyter_client
+  - jupyter_console
+  - jupyter_core
+  - jupyterlab
+  - jupyterlab_server
+  - keyring
+  - kiwisolver
+  - krb5
+  - lazy-object-proxy
+  - libarchive
+  - libcurl
+  - libffi
+  - liblief
+  - libpng
+  - libsodium
+  - libspatialindex
+  - libssh2
+  - libtiff
+  - libxml2
+  - libxslt
+  - llvmlite
+  - locket
+  - lxml
+  - lz4-c
+  - lzo
+  - markupsafe
+  - matplotlib
+  - matplotlib-base
+  - mccabe
+  - mistune
+  - mkl
+  - mkl-service
+  - mkl_fft
+  - mkl_random
+  - mock
+  - more-itertools
+  - mpc
+  - mpfr
+  - mpmath
+  - msgpack-python
+  - multipledispatch
+  - nbconvert
+  - nbformat
+  - networkx
+  - ninja
+  - nltk
+  - nose
+  - notebook
+  - numba
+  - numexpr
+  - numpy
+  - numpy-base
+  - numpydoc
+  - olefile
+  - openpyxl
+  - openssl
+  - packaging
+  - pandas
+  - pandoc
+  - pandocfilters
+  - pango
+  - parso
+  - partd
+  - path
+  - path.py
+  - pathlib2
+  - pathtools
+  - patsy
+  - pcre
+  - pep8
+  - pexpect
+  - pickleshare
+  - pillow
+  - pip
+  - pixman
+  - pkginfo
+  - pluggy
+  - ply
+  - prometheus_client
+  - prompt-toolkit
+  - prompt_toolkit
+  - psutil
+  - ptyprocess
+  - py
+  - py-lief
+  - pycodestyle
+  - pycosat
+  - pycparser
+  - pycrypto
+  - pycurl
+  - pydocstyle
+  - pyflakes
+  - pygments
+  - pylint
+  - pyodbc
+  - pyopenssl
+  - pyparsing
+  - pyqt
+  - pyrsistent
+  - pysocks
+  - pytables
+  - pytest
+  - pytest-arraydiff
+  - pytest-astropy
+  - pytest-astropy-header
+  - pytest-doctestplus
+  - pytest-openfiles
+  - pytest-remotedata
+  - python
+  - python-dateutil
+  - python-jsonrpc-server
+  - python-language-server
+  - python-libarchive-c
+  - pytorch
+  - pytz
+  - pywavelets
+  - pyxdg
+  - pyyaml
+  - pyzmq
+  - qdarkstyle
+  - qt
+  - qtawesome
+  - qtconsole
+  - qtpy
+  - regex
+  - requests
+  - ripgrep
+  - rope
+  - rtree
+  - ruamel_yaml
+  - scikit-image
+  - scikit-learn
+  - scipy
+  - seaborn
+  - send2trash
+  - setuptools
+  - simplegeneric
+  - singledispatch
+  - sip
+  - six
+  - snappy
+  - snowballstemmer
+  - sortedcollections
+  - sortedcontainers
+  - soupsieve
+  - sphinx
+  - sphinxcontrib
+  - sphinxcontrib-applehelp
+  - sphinxcontrib-devhelp
+  - sphinxcontrib-htmlhelp
+  - sphinxcontrib-jsmath
+  - sphinxcontrib-qthelp
+  - sphinxcontrib-serializinghtml
+  - sphinxcontrib-websupport
+  - spyder
+  - spyder-kernels
+  - sqlalchemy
+  - sqlite
+  - statsmodels
+  - sympy
+  - tbb
+  - tblib
+  - terminado
+  - testpath
+  - threadpoolctl
+  - tk
+  - toolz
+  - torchvision
+  - tornado
+  - tqdm
+  - traitlets
+  - typed-ast
+  - typing_extensions
+  - ujson
+  - unicodecsv
+  - urllib3
+  - watchdog
+  - wcwidth
+  - webencodings
+  - werkzeug
+  - wheel
+  - widgetsnbextension
+  - wrapt
+  - xlrd
+  - xlsxwriter
+  - xlwt
+  - xz
+  - yaml
+  - yapf
+  - zeromq
+  - zict
+  - zipp
+  - zlib
+  - zstd
   - pip:
-    - absl-py==0.9.0
-    - cachetools==4.1.0
-    - google-auth==1.18.0
-    - google-auth-oauthlib==0.4.1
-    - grpcio==1.29.0
-    - markdown==3.2.2
-    - nibabel==3.1.0
-    - oauthlib==3.1.0
-    - protobuf==3.12.2
-    - pyasn1==0.4.8
-    - pyasn1-modules==0.2.8
-    - pytorch-lightning==0.8.1
-    - requests-oauthlib==1.3.0
-    - rsa==4.6
-    - tensorboard==2.2.2
-    - tensorboard-plugin-wit==1.6.0.post3
-    - pybids==0.11.1
-prefix: /home/siddhesh/anaconda3/envs/deepbet
+      - wurlitzer
+      - secretstorage
+      - absl-py
+      - bids
+      - bids-validator
+      - cachetools
+      - docopt
+      - google-auth
+      - google-auth-oauthlib
+      - grpcio
+      - markdown
+      - nibabel
+      - num2words
+      - oauthlib
+      - protobuf
+      - pyasn1
+      - pyasn1-modules
+      - pybids
+      - pytorch-lightning
+      - requests-oauthlib
+      - rsa
+      - tensorboard
+      - tensorboard-plugin-wit
+prefix: C:\Users\username\anaconda3\envs\brainmage_noversion