Add JIT stuff for better performance