Lec 12 ex: improve function to change fig colors