Generate Z momentum in TrackGenModule