Updated plotting of subset