Move each mod into its own class