added UI functionality