Added depth param, need to use it in decsion tree