Separated network into separate class