class diagram is added in readme