adding in the inventory class