Improved matplotlib plotting ideas