Remove plotting from demo