Added unit tests to project (although they need rewriting)