add problems 4&5