Update tests
Added tests