Add doctrings