make code better

make code better