Added normalization to keep things stable, but it still seems unstable