Stationary states are now working and normalized