Add features generator in python to utilize existing working model