Remove test data
Remove test data
Clean up code