Almost have working Cube class