simplified draw function in square plot