Add return docs