Move checks into checks.py