Remove 'with' statements, modify code before compiling