Added HDE model support