Changed matrix sizes and plot slightly