Add problem3.py (incomplete)