- problem 3 & 4 updates