updated to work with api for new weights