added class diagram... final touch..