Improved get feature model function