Added code and plots