Created separate classes for model features