Updating scripts for paper 4
Updated plotting functions
Updated models