Added ui funcionality