Fixed code style after code review