start fixing pyplot code