nicer draw function 2