UI modification, add matrix size input