Add balance model
Fix model balance
Fix auto balance
Merge