Thank you so much! Last question and I will stop being a pain. When you say to do the model surgery before adding them as submodules you mean to do the surgery still in the constructor but before using add_module right?