Added momentum matrix and got eigenvalues.
Changed how code works. Added shape variables.