Update design_matrix.py