Adding functionality to support UI elements