Simplify code context selection