Finished up ProblemB