Setup unit testing
Bug fixes
Improve unit test
Use default script