I have written down a similar derivation here if anyone is interested: https://ngr.yt/blog/kalman/