add code for coefficient plot