Problem 6 added for Python
Update