added code and plots for hw1