Changed load_module so class gets reloaded as well.