A few tweaks to your 401(k) can make it better than a pension