correct pyplot graph output