Started work on the final diagonal check, also removed some print stmts