Create custom wizard step

Hello,


I'm currently doing a refactoring in the golang plugin so that we can set some global variables ahead of user creating the project.
While it works ok for non-IDEA IDEs (PHPStorm et. al), I don't really understand how I should change the "createWizardSteps" method from the "public class GoModuleType extends ModuleType<GoModuleBuilder>" in order to add a step in which the user is presented with a custom dialog and sets the 'GOPATH' (choose a folder).
You can view the work in progress PR here: https://github.com/go-lang-plugin-org/go-lang-idea-plugin/pull/659 and the class I'm mentioning https://github.com/go-lang-plugin-org/go-lang-idea-plugin/blob/master/src/ro/redeul/google/go/ide/GoModuleType.java
Any pointers please?


Thank you.