Clean up the line chart with the refactoring of the line chart