Adding in more code for the chart