small changes for code review