Add test to test basic functionality