Coding done, start doing unit tests