Adding some tests and a bit of code