fixed the plotting of head feature