updated package so it works with the new library