Refactor and Clean-up Code

[eliseoe]

The goal is to improve code readability, maintainability, and overall structure without introducing functional changes. This includes tasks such as:

• Improving code organisation and structure.
• Renaming variables/functions where appropriate for clarity.
• Removing redundant or unused code.
• Simplifying overly complex logic where possible.
• Improving comments and documentation where needed.
• Applying consistent formatting and coding conventions.

This work was discussed with Defne and is intended as a non-functional refactor to make the codebase easier to understand and maintain. No changes to the external API or model behaviour should be introduced.