how to customize the feature selection dialog in wix4

[sagizar]

Hi,
I'm implementing an installer using Wix version4 and need to add and customize the feature selection dialog.
I already implement the installer, but never had to change the UI.
First thing that I need to do is to add logic for dependency between several features.
so if I select FeatureA, I want to select automatically and that the GUI will show that FeatureB is selected. but its OK to select only FeatureB.
Any help is welcome.
Thanks,
Sagi