Program in python to design matrix