Add classes to be loaded in unittest