More flexibility on matrix size