Add missing imports
Add get data function
Fix incorrectly named variables