Python file for generating the matrix along with the sample input used for testing