Added code for problem set 4 of DSGA1003.