which mosfet

First determine voltage and current requirements, then calculate the maximum on-resistance you can
tolerate, then select a device that's logic level and with that low an on-resistance and high enough voltage.

I doubt a lego motor pulls more that 2A

your 9V battery is the weak link in the chain.