Updated driver script for problem 3