Add association code to admin controller