Completed Python Assignment