Data science

Move these three lines into a separate method to remove arrowhead.