changes made to unit testing