Use traits

A simpler way to check if something is a host or controller