diff --git a/alphafold_casp13/README.md b/alphafold_casp13/README.md
index 7ea411c..191626e 100644
--- a/alphafold_casp13/README.md
+++ b/alphafold_casp13/README.md
@@ -30,9 +30,26 @@ forked `deepmind_research` repository using:
 ```shell
 python3 -m venv alphafold_venv
 source alphafold_venv/bin/activate
+pip install wheel
 pip install -r alphafold_casp13/requirements.txt
 ```
 
+## Data
+
+While the code is licensed under the Apache License, the AlphaFold weights and
+data are made available for non-commercial use only under the terms of the
+Creative Commons Attribution-NonCommercial 4.0 International (CC BY-NC 4.0)
+license. You can find details at:
+https://creativecommons.org/licenses/by-nc/4.0/legalcode
+
+In order to download the AlphaFold weights and data, you will need to request
+access using the
+[request form](https://docs.google.com/forms/d/1yrZXhQfSlwYnouDujrL2RkZKVBjF5AjomyF_RJ95dew/).
+
+Once you have obtained access, you can download the data from
+[Google Cloud Storage](https://console.cloud.google.com/storage/browser/alphafold_casp13_data).
+
+
 ### Input data
 
 The dataset to reproduce AlphaFold's CASP13 results can be downloaded from
diff --git a/alphafold_casp13/contacts_network.py b/alphafold_casp13/contacts_network.py
index 6cb00c3..c911df3 100644
--- a/alphafold_casp13/contacts_network.py
+++ b/alphafold_casp13/contacts_network.py
@@ -55,7 +55,6 @@ class ContactsNet(sonnet.AbstractModule):
                resolution_noise_scale,
                scalars,
                targets,
-               network_2d,
                network_2d_deep,
                torsion_bins=None,
                skip_connect=0,
@@ -88,7 +87,6 @@ class ContactsNet(sonnet.AbstractModule):
     self._torsion_bins = torsion_bins
     self._skip_connect = skip_connect
     self._targets = targets
-    self._network_2d = network_2d
     self._network_2d_deep = network_2d_deep
 
     self.asa_multiplier = asa_multiplier