make code better for vim

make code better for vim