Add functionality added on UI