improve documentation

README.md

 # Reworked Implementation of RLN
 
-This is the 2d version of the original implementation.
+This is the 3d version of the tf2 implementation.
+
+## Missing:
+- a way to read images 
+- never tested
+
 
 Original Paper:
 
@@ -10,11 +15,12 @@ Original Paper:
 
 This repository has three main branches:
 
-|    Name    | Explanation                                                                                                                                                                                                                    |
-|:----------:|:-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
-|    main    | The original code from the paper with minor bug fixes  & quality of life improvements.                                                                                                                                         |
-| **2d-tf2** | This branch is a 2d implementation of the same network design, but in TensorFlow 2. It is still based on the code from the original paper and uses parts of the original logic. This contains the main results of the project. |
-|   3d-tf2   | This branch contains the same network as of 2d-tf2, but the functions are replaced with their 3d equivalents. **Untested and still needs work.**                                                                               |
+|     Name      | Explanation                                                                                                                                                                                                                    |
+|:-------------:|:-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
+|     main      | The original code from the paper with minor bug fixes  & quality of life improvements.                                                                                                                                         |
+|   2d-tf2    | This branch is a 2d implementation of the same network design, but in TensorFlow 2. It is still based on the code from the original paper and uses parts of the original logic. This contains the main results of the project. |
+|  **3d-tf2**   | This branch contains the same network as of 2d-tf2, but the functions are replaced with their 3d equivalents. **Untested and still needs work.**                                                                               |
+
 
 ## Requirements