Added functionality to display matrix A and matrix B, based on user input.