Originally I thought you had to match the top of the cube with the colour of the other cube, I didn't know you had to touch the side with the matching colour. Some clarification would be useful.
Aside from that, add collision/walls in order to stop you from rolling infinitely to make the puzzle more challenging and interesting.