Switching to tests. It is easier to read and to implement new cases.