Made more unit tests and doc strings