Fix: move model to features directory

correction to git conflict