Add separate class for UI methods