[MRG] specify axes in utils.draw