Add some code for demo installations