Updated classdiagram