UI functionality added