Wrap it up in a nice class