Add (incomplete) problem 6 python file