Updated program to display parabola