Add basic functional test