Added ability to load in sample data