improving docstrings in cpu.py