Looking good, we just need to add a couple more tests