Break down the pre_process function

[chinyeungli]

The current pre_process() is doing way to many things. We should break down these into couple of small functions instead.