Refactored code
updating plots
updating plots