accept end conditions for problems 1 and 2 as a parameter