Code updated
Added additional plots
Updated code
Updated README