Add more functions
Add functions
Update test script