Added one more function and it's test, this is done I believe