adding tests

Adding tests to this big project