Improved problem 6 Python solution, finished problems 7 and 8 in Python