Added tests for program