Almost correct calculation (some eigen states can be missed), plotting added