Recording maschine midi from keyboard mode?

[Double A to the Ron]

I'm trying to run maschine as a vst in Ableton and record midi from maschines keyboard mode. Ie. Load a kit in maschine and change pitch of a sample in that kit with keyboard mode and have that recorded into ableton.
Thanks for any help