Code cleanup
add network class diagram