mm: pte: Add comments and perf tweaks to pte switch

Mostly done some comments, but also restructured some of the code.
Nothing worth writing home about, besides the comments.

Signed-off-by: Bart Kuivenhoven <bemkuivenhoven@gmail.com>