Experiment 3. Updated next state to return rows syntax as in unit test