diff --git a/festim/hydrogen_transport_problem.py b/festim/hydrogen_transport_problem.py
index c5172441e..471e42e4d 100644
--- a/festim/hydrogen_transport_problem.py
+++ b/festim/hydrogen_transport_problem.py
@@ -139,7 +139,7 @@ def defing_export_writers(self):
 
     def define_function_space(self):
         """Creates the function space of the model, creates a mixed element if
-        model is multispecies"""
+        model is multispecies. Creates the main solution and previous solution function u and u_n."""
         element_CG = basix.ufl.element(
             basix.ElementFamily.P,
             self.mesh.mesh.basix_cell(),