Removing the debug px pz output.
Optimizing imports.