Add conditional compilation