changed some function to be protected as they should
added class diagram