Breaking files into a seperate class