The demo is really interesting already, with more upgrade options available it would definitely be great. One thing I'd add on top of that is maybe showing where enemy pieces will go on their next move when you hover over them or something (similar to Pawnbarian).