Improve error handling and logging in code execution tool