minor change to make the code into better style