Adding unit tests...