Add docs for model part and fix missing requires