Adding Automated Test