This should be a docstring, but otherwise looks good