Is there a case for having more encapsulation? So a class and tests are defined and the LLM only works on that.