Limited to two rows

Some test code to limit to two rows is removed. Fixed