Add derivative function
Fix import bug
Relocate squad in imports