updated tests. still passing all tests