Updated paper and code

Made revisions to paper. Nearly complete