User prompt to produce random walk in either 2D or 3D

I'm creating a program to simulate a random walk and have so far produced serparate code for it in both 2D and 3D. I'd like to give a user option at the beginning to do the walk in either 2D or 3D - how would I go about this?