updated for cleaner plot