Update feature to match user interface