Updated some code for the inventory

^ that