I've done the same, but in case you didn't know Kitty can do that automatically (though it does a bit more that you may or may not want it to do)

https://sw.kovidgoyal.net/kitty/kittens/ssh/