add cosmetic fixes