Add problem 2 for unit 1
Add problem 3 in unit 1