Add unit tests
Add unit testing to built script