Remove unused use statement
Add tests