Remove test code from demo