Add docs and make function name clearer