Functionality works now