add wrapper around plot