finished added the tests