improve feature

improve feature