Minor changes
Add Problem 5