Add some cube logic