Initial commit: Code from the paper