Updates to the code for the cube