Working version.... no AI yet!