support for animation plot from the 1st assignment