Add classes to controllers