#21 Added a few more test cases and a dummy test

that deletes all the files created from testing
#21 Restructured code to work with unit test