Organizing object-oriented code in Python