Added new test, should be working