Clean up source code of UIBehavior
Implement UIBehavior