Implemented the mix_columns() method