Updated code for data figures and figure 4
Updated abstract, edited some notation