Cosmetic. Tweaked some code to look better.