adding in unit tests