Update check_assignments.py

WTFF