Here is with the constant hot red from the 3-wire to the receptacle added back in. Just for ease of drawing, the constant is on the top, although yours is opposite.
