test: Add more test