Typo and small enhancement