Problem 5
Problem 6
Remove unnecessary code
Solved problem 7, 8, 9