minor change on code appearance