Improve the feature