Added classes for saving + loading

Changed how we're going to save and load by using classes instead of
methods because it's much cleaner.