slight code edit
3 files changed