Minor changes in matrix and image