Finished assignment

Only tested with
1 2 3
0 1 4
5 6 0
matrix. Could add more tests and print statements to make sure it's
working correctly.