Adds test for rows, columns, and diagonals and refactors each