This class will need to be made package private if it goes into `main` instead of `test`. I can do this when merging, just putting a reminder here.