Updated to work with most model types

still a bit dodgy and renders the shrouding.