Updated with functionality to remove product