Problems 3 and 4 in python
Adding Problem 6