finish unittest problem 1