Add Problem 3
Minor changes