a bit more polish
2 files changed