diff --git a/alphafold_casp13/two_dim_convnet.py b/alphafold_casp13/two_dim_convnet.py
index 229663b..1914098 100644
--- a/alphafold_casp13/two_dim_convnet.py
+++ b/alphafold_casp13/two_dim_convnet.py
@@ -12,7 +12,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-"""One dim convnet for angles prediction."""
+"""Two dimensional convolutional neural net layers."""
 
 from absl import logging
 import tensorflow.compat.v1 as tf