Add a helper for CPU normalization
Fix some random pyflakes warnings