Extensive edits to the plotting routine - now plots all staircases