adding in changes to unit testing