Added couple of lines

Cosmetic change