adds more unit tests