ENH: Change figure size
RF/ENH: Move matplotlib imports into plotting function
BF: Deal with NaNs, convert timeline to integer index values