Updating UI, adding user functionality