edited prob4