Small code improvements.

- Refactor of case_study variable to json_data
- Add context to some methods