Python Add Function To Existing Class