Add assertion on y equal +1 or -1 AND remove plots of problem 1