Adding code and plot image to repo