More updates to the paper