Updated demo main()

The code for the shift register will be added in later