remove unused import
add some tests