Make cube draw function more generic (any size cube) and move into base class