Fixed bug in screen plots