Add test and method for setRows
Add setCols function and a test for it.
Add tests for the checkLocked function.