Adding tests cases for main program