added a test and make sure it passes