Updated Arxiv paper