Update check_allocation.py

Improved formatting.