You may have disrupted sediment when you swapped out the water heater, and debris could be in your shower faucet, and in the faucet screens. I would disassemble your shower faucet first and clean the guts. Those things are supposed to regulate temperature and pressure to prevent scalding, and if there's gunk in there, it cannot do it's job