Update channel_assignment.py