fixed box width and height bug and also some figures for demonstration