added unit test project to solution and changed class diagram