updated test to be more comprehensive