Improved code, more colors and a cube