adding exploratory code and plots