Added comments and removed network class