I do the same. I spend a lot of time sketching and mentally toying with it before I drop code. It gives me something to do on train rides and is a welcome luxury in comparison to agile.