Fixed up two plotting scripts