Refactored traffic module to support easier behavior subclassing