add assumptions to problem 3