updaed to have width size correct