Add unit controller spec