Awesome, thanks!
So I made some tweaks and merged into my code, so that it works for many models now. Here you can find the generic version: