fix review issues (commas, docstrings etc)