Added framework code for demo setup