Initial commit of the code

First commit of the code itself. Also updated the
readme to refer to the correct arXiv publication.