Added mod functionality