changed test name
Cleaned up imports