updated test to reflect correct answer and figured out solution