finished up unit tests