Add docs to models