Ok, first remove the flared part of the flashlight (like you were going to change the bulb) and the tailcap. Stick an inkpen or something similar into the battery compartment (there is a plastic piece you will need to hold in place with the inkpen). Pull the bulb out. There is a black piece of plastic that the bulb went into that says "Do not remove" (of course, we must now remove it). Using tweezers or a knife, gently pry this plastic part off while holding the inkpen firmly against the end of the battery compartment. You will now be able to see the plastic piece that your inkpen is holding in place. This plastic part will tell you where to put your positive and negative wires. When you look at the plastic part, you will see a metal tab that makes contact with the edge of the hole you are looking through. This is your negative contact. It ends near the center of the plastic part you are holding in place. That end point is where your bulb made its negative contact. Take note of which side it is on, then gently replace the other plastic part that you took off. Feed a wire into the hole in that plastic part that corresponds with your negative contact. The other end of this wire goes to the outside of your connector for the atomizer (the part usually salvaged from a dead e-cig battery). Feed another wire into the other hole. This is your positive wire. The other end will go to the center of your atomizer connector. In most mods we would put a switch in the positive line, but you have stated you want a simple twist on like the flashlight already had. So, no switch, just a straight shot from the battery to the atomizer. Replace your tailcap as it is part of the negative side of the circuit. Mount your atomizer connection as you had planned. Reassemble flashlight and enjoy.
Here is my two cents on the battery issue. Two AA batteries will produce some vapor with a 901 atomizer, but not efficiently, quickly, or abundantly. It will take more time to heat up the atomizer than the original 901 battery did. You can get a lithium battery the same size. It is called 14500. You cannot use two of the lithium batteries with it though. Two 14500 batteries stacked like that is about 7.2 volts. I have tried this with 901 atomizers and it kills them dead instantly. If you go for the lithium battery, you will need to use a spacer of some sort (rare earth magnets work nicely) to fill the space of one battery (50mm).