Clean up code samples/demo