Implement restock function