Refactoring to add a component behavior. Easy pluggable custom UI