Update Problem-3.py

Should probably remove old methods, and fix broken one