Re-added System.Diagnostics dependency