setted up missing part to finish model testing