updated test to new features