light format fixes
diff --git a/docs/docs/frameworks.md b/docs/docs/frameworks.md
index ffb48c0..a0c787c 100644
--- a/docs/docs/frameworks.md
+++ b/docs/docs/frameworks.md
@@ -169,10 +169,13 @@
```python
from amaterasu.pyspark.runtime import AmaContext
+
ama_context = AmaContext.builder().build()
dataset_conf = ama_context.dataset_manager.get_dataset_configuration("mygenericdataset") # This is a dataset without builtin support
+
some_prop = dataset_conf['mydataset_prop1']
some_prop2 = dataset_conf['mydataset_prop2']
+
my_df = my_custom_loading_logic(some_prop, some_prop2)
my_new_df = black_magic(my_df)
ama_context.persist('magic_df', my_new_df) # This is a dataset with builtin support