Added class diagram Added tests for main