Added small cosmetic fix to code