updated tests based on test chapter