Python Pass Class Reference In Functions