Add basic functionality tests