started adding in all the unit tests