Place demo assembly into its own module