Added basic unit tests for script