Did unit tests on code, added to comments.