Would be good to put :return: and :rtype:.