Updating the prove script