refactor tests and add a new test case that tests 2 x 2 word squares
remove the need for a(index interator) and instead use words.size which functions as the same thing