Added controller assignment

Oops