Update vm.rs

Moved the feature gated logic within `vm.rs` to macros as a readability test.
Moving the logic to a function caused borrow errors.

Not sure if this will be kept but its worth testing.