added tests for all the code that I wrote