Problem 4

Added Problem 4 and added problem in comments at the top of the class