This is mainly about refactoring the codebase and adding tests that are currently missing. We should also keep an eye on grouping by feature during this process as this allows for better scaling in the long run.