Save after class

From here: DO A TABLE JOIN!
This is the simple way, instead of elif elif elif