imporve the code style