Remove unused class in demo project