Slight changes

Gonna make a new branch to redo network code