Unit tests

Created some unit tests