Adding code to grab data