Further polish
2 files changed