Updated arg names in problem3