Python Use Class Type Inside A Shape