Changed everything. Put the save method in the classes because it is smarter yeah.