Update to a separate gem