Put everything in a class