Python Add Class Argument To Function Normally