Update wedding.py

Update wedding.py to be closer to what the homework solution says