Add problem6
Fix problem4