correct functions to run the unit tests
to specifications
add functions for last two unit tests