wrote more unit tests