added code for coefficient plot