minor code styling update