Add automated script to test the code change