Update cpu.py

Minor updates on comments