- small adjustment on problem3