Adding code and plots