bug fixes
Add KA and GA at the model function