diff --git a/.github/workflows/test_suite.yaml b/.github/workflows/test_suite.yaml
index e356f77c..80f7b044 100644
--- a/.github/workflows/test_suite.yaml
+++ b/.github/workflows/test_suite.yaml
@@ -68,11 +68,20 @@ jobs:
 
       - name: Run Unit Tests
         run: |
-          python -m pytest -n 2 -l ${{ env.extra_pytest_args }} --ignore tests/test_integration_features.py --cov=py21cmfast --cov-config=.coveragerc -vv --cov-report xml:./coverage.xml --durations=25
+          python -m pytest -n 2 -l ${{ env.extra_pytest_args }} \
+          --ignore tests/test_integration_features.py \
+          --cov=py21cmfast --cov-config=.coveragerc \
+          -vv --cov-report xml:./coverage.xml --durations=25
+
 
       - name: Run Integration Tests
         run: |
-          python -m pytest -n 2 -l ${{ env.extra_pytest_args }} tests/test_integration_features.py --cov=py21cmfast --cov-config=.coveragerc -vv  --cov-append --cov-report xml:./coverage.xml --durations=25 --plots=testplots
+          python -m pytest -n 2 -l ${{ env.extra_pytest_args }} \
+            tests/test_integration_features.py \
+            --cov=py21cmfast --cov-config=.coveragerc \
+            -vv  --cov-append --cov-report xml:./coverage.xml \
+            --durations=25 --plots=testplots \
+            --junitxml=junit.xml -o junit_family=legacy
 
       - name: Archive Integration Test Plots
         if: always()
@@ -88,3 +97,9 @@ jobs:
           fail_ci_if_error: true
           verbose: true
           token: ${{ secrets.CODECOV_TOKEN }}
+
+      - name: Upload test results to Codecov
+        if: ${{ !cancelled() }}
+        uses: codecov/test-results-action@v1
+        with:
+          token: ${{ secrets.CODECOV_TOKEN }}
diff --git a/README.rst b/README.rst
index ad67bd10..88d87c54 100644
--- a/README.rst
+++ b/README.rst
@@ -3,27 +3,31 @@
 ========
 
 .. start-badges
-.. image:: https://travis-ci.org/21cmFAST/21cmFAST.svg
-    :target: https://travis-ci.org/21cmFAST/21cmFAST
-.. image:: https://coveralls.io/repos/github/21cmFAST/21cmFAST/badge.svg
-    :target: https://coveralls.io/github/21cmFAST/21cmFAST
-.. image:: https://img.shields.io/badge/code%20style-black-000000.svg
-    :target: https://github.com/ambv/black
+.. image:: https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/ruff/main/assets/badge/v2.json
+    :target: https://github.com/21cmFAST/21cmFAST
+.. image:: https://img.shields.io/pypi/v/21cmFAST.svg
+    :target: https://pypi.org/pypi/21cmFAST
+.. image:: https://img.shields.io/pypi/l/21cmFAST.svg
+    :target: https://github.com/21cmFAST/21cmFAST/blob/main/LICENSE
+.. image:: https://codecov.io/gh/21cmfast/21cmFAST/graph/badge.svg?token=sPc47SaC7Y
+    :target: https://codecov.io/gh/21cmfast/21cmFAST
+.. image:: https://img.shields.io/pypi/pyversions/21cmFAST.svg
+    :target: https://pypi.python.org/pypi/21cmFAST
+.. image:: https://github.com/21cmfast/21cmFAST/actions/workflows/test_suite.yaml/badge.svg
+    :target: https://github.com/21cmfast/21cmFAST/actions/workflows/test_suite.yaml
 .. image:: https://readthedocs.org/projects/21cmfast/badge/?version=latest
     :target: https://21cmfast.readthedocs.io/en/latest/?badge=latest
-    :alt: Documentation Status
+    :alt: Documentation
 .. image:: https://img.shields.io/conda/dn/conda-forge/21cmFAST
     :target: https://github.com/conda-forge/21cmfast-feedstock
     :alt: Conda
 .. image:: https://joss.theoj.org/papers/10.21105/joss.02582/status.svg
-   :target: https://doi.org/10.21105/joss.02582
+    :target: https://doi.org/10.21105/joss.02582
 .. end-badges
 
 **A semi-numerical cosmological simulation code for the radio 21-cm signal.**
 
-.. image:: joss-paper/yuxiangs-plot-small.png
-    :target: http://homepage.sns.it/mesinger/Media/lightcones_minihalo.png
-
+.. image:: docs/images/angular_lightcone_horizontal.PNG
 
 This is the official repository for ``21cmFAST``: a semi-numerical code that is able to
 produce 3D cosmological realisations of many physical fields in the early Universe.
diff --git a/docs/images/angular_lightcone_horizontal.PNG b/docs/images/angular_lightcone_horizontal.PNG
new file mode 100644
index 00000000..4ff48646
Binary files /dev/null and b/docs/images/angular_lightcone_horizontal.PNG differ