Updated paper, cleaned repo to be just code.