started adding in unit tests
added more unit tests