Update feature for new design