Set up the mod class