Test cases need to done and output to be put at end of code.