add some tests
fix bug with removing null keys
add test to increase coverage